print('='*30,'第一章作业2','='*30)
print()

import os

def dirsiz(di):
    sum=0
    # 判断目录是否为文件
    if os.path.isfile(di):
        sum+=os.path.getsize(di)
    # 判断指定目录是否为目录
    if os.path.isdir(di):
        dirLi=os.listdir(di)
        for i in dirLi:
            # 拼接目录下文件或者文件夹的地址
            file=os.path.join(di,i)
            if os.path.isdir(file):
                #递归统计
                sum+=dirsiz(file)
            if os.path.isfile(file):
                sum+=os.path.getsize(file)
    return sum

while True:
    add_r=input("输入文件目录:")
    dirsiz(add_r)
    print("{}字节".format(dirsiz(add_r)))
    input("按回车键继续：")